from django.shortcuts import render

# Create your views here.
from django.views.generic import View
from goods.models import GoodsCategory,GoodsChannel
from contents.models import ContentCategory
from collections import OrderedDict

# Create your views here.
class IndexView(View):
    """
    获取首页分类数据

    GET /goods/categories/
    """
    def get(self,request):
        # 使用有序字典保存类别的顺序
        # categories = {
        #     1: { # 组1
        #         'channels': [{'id':, 'name':, 'url':},{}, {}...],
        #         'sub_cats': [{'id':, 'name':, 'sub_cats':[{},{}]}, {}, {}, ..]
        #     },
        #     2: { # 组2
        #
        #     }
        # }
        # 初始化存储容器
        categories = OrderedDict()
        # 获取一级分类
        channels = GoodsChannel.objects.order_by('group_id', 'sequence')

        # 对一级分类进行遍历
        for channel in channels:
            # 获取group_id
            group_id = channel.group_id
            # 判断group_id 是否在存储容器,如果不在就初始化
            if group_id not in categories:
                categories[group_id] = {
                    'channels': [],
                    'sub_cats': []
                }

            one = channel.category
            # 为channels填充数据
            categories[group_id]['channels'].append({
                'id': one.id,
                'name': one.name,
                'url': channel.url
            })
            # 为sub_cats填充数据
            for two in one.goodscategory_set.all():
                # 初始化 容器
                two.sub_cats = []
                # 遍历获取
                for three in two.goodscategory_set.all():
                    two.sub_cats.append(three)

                # 组织数据
                categories[group_id]['sub_cats'].append(two)

        # 广告和首页数据
        contents = {}
        content_categories = ContentCategory.objects.all()
        # content_categories = [{'name':xx , 'key': 'index_new'}, {}, {}]

        # {
        #    'index_new': [] ,
        #    'index_lbt': []
        # }
        for cat in content_categories:
            contents[cat.key] = cat.content_set.filter(status=True).order_by('sequence')

        # 组织上下文进行模板渲染
        context = {
            'categories': categories,
            'contents': contents
        }
        return render(request, 'index.html', context)


# APIView
# GenericAPIView
# CreateAPIView
from rest_framework.generics import GenericAPIView,ListAPIView
from goods.models import SKU
from users.serializers import SKUSerializer
from rest_framework.response import Response


class HotSKUView(ListAPIView):
    def get_queryset(self):
        category_id = self.kwargs['category_id']
        return SKU.objects.filter(category_id=category_id,is_launched=True).order_by('-sales')[:2]

    serializer_class = SKUSerializer
    pagination_class = None

    """
    GET      /goods/categories/(?P<category_id>\d+)/hotskus/

    1.根据用户选择的分类id，去获取商品信息，商品信息都是模型，返回JSON
    2.根据销量进行排序
    """

    # def get_queryset(self):
    #     category_id = self.kwargs['category_id']
    #     return SKU.objects.filter(category_id=category_id,is_launched=True).order_by('-sales')[:2]
    #
    # serializer_class = SKUSerializer
    #
    # def get(self, request, category_id):
    #
    #     # 1.获取数据
    #     skus = self.get_queryset()
    #     # 2.创建序列化器
    #     serializer = self.get_serializer(skus,many=True)
    #
    #     return Response(serializer.data)

# APIView
# GenericAPIView
# CreateAPIView
from rest_framework.filters import OrderingFilter
from rest_framework.pagination import LimitOffsetPagination
# from utils.pagination import CustomPageNumberPagination
class SKUListView(ListAPIView):

    """

    GET /goods/categories/(?P<category_id>\d+)/skus/?ordering=xxxx&page=xxx&page_size=xxx

    1. 我们根据商品的分类来获取商品的信息, 最终肯定是要返回JSON
    2. 我们需要对数据进行排序处理
    3. 我们需要对数据进行分页处理
    """

    # queryset = SKU.objecst.all()
    serializer_class = SKUSerializer
    def get_queryset(self):

        category_id = self.kwargs['category_id']
        return SKU.objects.filter(category_id=category_id,is_launched=True)

    #添加排序的代码
    filter_backends = [OrderingFilter]
    ordering_fields = ['create_time','price','sales']


    # pagination_class = CustomPageNumberPagination

from .serializers import SKUIndexSerializer
from drf_haystack.viewsets import HaystackViewSet

class SKUSearchViewSet(HaystackViewSet):
    """
    SKU搜索
    """
    index_models = [SKU]

    serializer_class = SKUIndexSerializer














